import time
from basepage import BasePage
from Python基础软件测试课程.APP自动化.image_recognition import ImageRecognition
class Screen(BasePage):
    # 图片验证码输入框
    _img_check_code = "/hierarchy/android.widget.FrameLayout/android.widget.LinearLayout/android.widget.FrameLayout/android.widget.LinearLayout/android.widget.FrameLayout/android.widget.LinearLayout/android.widget.FrameLayout/android.widget.LinearLayout/android.widget.ScrollView/android.widget.RelativeLayout/android.widget.RelativeLayout[3]/android.widget.ImageView[1]"
    # 截取页面中特定区域的图片
    def get_part_screen(self):
        # 截图将图片保存至固定的位置
        img_folder = 'E:\\Game\\image'
        # 截图的文件名组成
        times = time.strftime('%Y%m%d%H%M', time.localtime(time.time()))
        # 图片存储位置+文件名
        screen_save_path = img_folder + times + '.png'
        # 截取特定位置的图片，并保存在已定义的存储位置和规定的文件名
        self.find_xpath(self._img_check_code).screenshot(screen_save_path)
        # 实例化ImageRecognition，传入需要识别的图片位置
        ir = ImageRecognition(screen_save_path)
        a = ir.ocr()
	return a
